// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef NET_UDP_FUZZED_DATAGRAM_CLIENT_SOCKET_H_
#define NET_UDP_FUZZED_DATAGRAM_CLIENT_SOCKET_H_

#include "net/udp/datagram_client_socket.h"

#include <stdint.h>

#include "base/memory/weak_ptr.h"
#include "net/base/completion_callback.h"
#include "net/base/ip_endpoint.h"
#include "net/base/network_change_notifier.h"
#include "net/log/net_log.h"

namespace net {

class FuzzedDataProvider;
class IOBuffer;

// Datagram ClientSocket implementation for use with fuzzers. Can fail to
// connect, reads and writes can succeed or fail synchronously or
// asynchronously.  Successful reads return random data.
class FuzzedDatagramClientSocket : public DatagramClientSocket {
public:
    // |data_provider| must outlive the created socket.
    explicit FuzzedDatagramClientSocket(FuzzedDataProvider* data_provider);
    ~FuzzedDatagramClientSocket() override;

    // DatagramClientSocket implementation:
    int Connect(const IPEndPoint& address) override;
    int ConnectUsingNetwork(NetworkChangeNotifier::NetworkHandle network,
        const IPEndPoint& address) override;
    int ConnectUsingDefaultNetwork(const IPEndPoint& address) override;
    NetworkChangeNotifier::NetworkHandle GetBoundNetwork() const override;

    // DatagramSocket implementation:
    void Close() override;
    int GetPeerAddress(IPEndPoint* address) const override;
    int GetLocalAddress(IPEndPoint* address) const override;
    const BoundNetLog& NetLog() const override;

    // Socket implementation:
    int Read(IOBuffer* buf,
        int buf_len,
        const CompletionCallback& callback) override;
    int Write(IOBuffer* buf,
        int buf_len,
        const CompletionCallback& callback) override;
    int SetReceiveBufferSize(int32_t size) override;
    int SetSendBufferSize(int32_t size) override;

private:
    void OnReadComplete(const net::CompletionCallback& callback, int result);
    void OnWriteComplete(const net::CompletionCallback& callback, int result);

    FuzzedDataProvider* data_provider_;

    bool connected_ = false;
    bool read_pending_ = false;
    bool write_pending_ = false;

    BoundNetLog net_log_;

    IPEndPoint remote_address_;

    base::WeakPtrFactory<FuzzedDatagramClientSocket> weak_factory_;

    DISALLOW_COPY_AND_ASSIGN(FuzzedDatagramClientSocket);
};

} // namespace net

#endif // NET_UDP_FUZZED_DATAGRAM_CLIENT_SOCKET_H_
